"""
RFdiffusion配置文件
"""

# 从环境变量读取配置
import os

# 图后端选项：'dgl'或'pyg'
# 使用函数而不是属性，以便在每次访问时都检查环境变量
def get_graph_backend():
    """从环境变量或默认值获取图后端设置"""
    # 有效的后端选项
    VALID_BACKENDS = ['dgl', 'pyg']
    
    # 从环境变量获取后端设置
    backend = os.environ.get('RFDIFFUSION_GRAPH_BACKEND', 'dgl').lower()
    
    # 验证后端选项
    if backend not in VALID_BACKENDS:
        print(f"警告：无效的图后端'{backend}'，使用默认值'dgl'")
        return 'dgl'
    
    return backend

# 设置默认值
GRAPH_BACKEND = get_graph_backend() 